////////////////////////////////////////////////
// DESCRIPTION:
//    MSHelp2 registration support
//
// Legal Notices:
//   Copyright (c) 2008, Telliam Consulting, LLC.
//   All rights reserved.
//
//   Redistribution and use in source and binary forms, with or without modification,
//   are permitted provided that the following conditions are met:
//
//   * Redistributions of source code must retain the above copyright notice, this list
//     of conditions and the following disclaimer.
//   * Redistributions in binary form must reproduce the above copyright notice, this
//     list of conditions and the following disclaimer in the documentation and/or other
//     materials provided with the distribution.
//   * Neither the name of Telliam Consulting nor the names of its contributors may be
//     used to endorse or promote products derived from this software without specific
//     prior written permission.
//
//   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
//   EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
//   OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
//   SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
//   INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
//   TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR 
//   BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
//   CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 
//   ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
//   DAMAGE. 
//
#pragma once
#include "CustomActionHelpers.h"

class MsHelp2Session
{
    MSHelpServices::IHxRegisterSessionPtr pRegistrationSession;

public:    
    MsHelp2Session()
        : pRegistrationSession(__uuidof(MSHelpServices::HxRegisterSession))
    {
    }
   
    CString CreateTransaction()
    {
        _bstr_t txid;
        return (LPCTSTR)(this->pRegistrationSession->CreateTransaction( txid ));
    }

    void ContinueTransaction(CString TxID)
    {
        this->pRegistrationSession->ContinueTransaction( _bstr_t( TxID.AllocSysString(), false) );
    }
    
    CString PostPoneTransaction()
    {
        return (LPCTSTR)(this->pRegistrationSession->PostponeTransaction());
    }

    void RollbackTransaction()
    {
        this->pRegistrationSession->RevertTransaction();
    }

    void CommitTransaction()
    {
        this->pRegistrationSession->CommitTransaction();
    }

    void RegisterNamespace(CString Namespace, CString Collection, CString Description)
    {
        MSHelpServices::IHxRegisterPtr pRegister = this->pRegistrationSession->GetRegistrationObject(MSHelpServices::HxRegisterSession_IHxRegister);
        pRegister->RegisterNamespace( _bstr_t(Namespace.AllocSysString(), false)
                                    , _bstr_t(Collection.AllocSysString(), false)
                                    , _bstr_t(Description.AllocSysString(), false)
                                    );
    }

    void RemoveNamespace(CString Namespace)
    {
        MSHelpServices::IHxRegisterPtr pRegister = this->pRegistrationSession->GetRegistrationObject(MSHelpServices::HxRegisterSession_IHxRegister);
        pRegister->RemoveNamespace( _bstr_t(Namespace.AllocSysString(), false) );
    }

    void RegisterHelpFile(CString Namespace, CString Id, int LangID, CString FileName)
    {
        MSHelpServices::IHxRegisterPtr pRegister = this->pRegistrationSession->GetRegistrationObject(MSHelpServices::HxRegisterSession_IHxRegister);
        pRegister->RegisterHelpFile( _bstr_t(Namespace.AllocSysString(), false)
                                   , _bstr_t(Id.AllocSysString(), false)
                                   , LangID
                                   , _bstr_t(FileName.AllocSysString(), false)
                                   );
    }

    void RemoveHelpFile(CString Namespace, CString Id, int LangID)
    {
        MSHelpServices::IHxRegisterPtr pRegister = this->pRegistrationSession->GetRegistrationObject(MSHelpServices::HxRegisterSession_IHxRegister);
        pRegister->RemoveHelpFile( _bstr_t( Namespace.AllocSysString(), false)
                                 , _bstr_t( Id.AllocSysString(), false)
                                 , LangID
                                 );
    }

    void RegisterHelpPlugIn( CString ProductNamespace, CString ProductHxt, CString Namespace, CString Hxt, CString Hxa, long options = 0 )
    {
        MSHelpServices::IHxPlugInPtr pPlugIn = this->pRegistrationSession->GetRegistrationObject(MSHelpServices::HxRegisterSession_IHxPlugIn);
        pPlugIn->RegisterHelpPlugIn( _bstr_t(ProductNamespace.AllocSysString(), false)
                                   , _bstr_t(ProductHxt.AllocSysString(), false)
                                   , _bstr_t(Namespace.AllocSysString(), false)
                                   , _bstr_t(Hxt.AllocSysString(), false)
                                   , _bstr_t(Hxa.AllocSysString(), false)
                                   , 0 );
    }

    void RemoveHelpPlugIn(CString ProductNamespace, CString ProductHxt, CString Namespace, CString Hxt, CString Hxa)
    {
        MSHelpServices::IHxPlugInPtr pPlugIn = this->pRegistrationSession->GetRegistrationObject(MSHelpServices::HxRegisterSession_IHxPlugIn);
        pPlugIn->RemoveHelpPlugIn( _bstr_t(ProductNamespace.AllocSysString(), false)
                                 , _bstr_t(ProductHxt.AllocSysString(), false)
                                 , _bstr_t(Namespace.AllocSysString(), false)
                                 , _bstr_t(Hxt.AllocSysString(), false)
                                 , _bstr_t(Hxa.AllocSysString(), false)
                                 );
    }

    void MergeIndex(CString Namespace);
};
